package entity;

import lombok.Data;

import java.util.HashMap;
import java.util.Map;

/**
 * Created with IntelliJ IDEA.
 *
 * @Author: torlesse-liang
 * @Date: 2022/04/16/0:51
 * @Description: 全局统一返回结果
 */
@Data
public class R {

    private Boolean success;

    private Integer code;

    private String message;

    private Map<String, Object> data = new HashMap<String, Object>();

    public R() {
    }

    /**
     * 功能描述：成功返回
     *
     * @return R 成功返回结果
     * @description:
     */
    public static R ok() {
        R r = new R();
        r.setSuccess(ResultCodeEnum.SUCCESS.getSuccess());
        r.setCode(ResultCodeEnum.SUCCESS.getCode());
        r.setMessage(ResultCodeEnum.SUCCESS.getMessage());
        return r;
    }

    /**
     * 功能描述：失败返回
     *
     * @return R 失败返回结果
     */
    public static R error() {
        R r = new R();
        r.setSuccess(ResultCodeEnum.UNKNOWN_REASON.getSuccess());
        r.setCode(ResultCodeEnum.UNKNOWN_REASON.getCode());
        r.setMessage(ResultCodeEnum.UNKNOWN_REASON.getMessage());
        return r;
    }

    /**
     * 功能描述：设置返回结果
     *
     * @param resultCodeEnum 结果枚举
     * @return R 返回给前端的结果
     */
    public static R setResult(ResultCodeEnum resultCodeEnum) {
        R r = new R();
        r.setSuccess(resultCodeEnum.getSuccess());
        r.setCode(resultCodeEnum.getCode());
        r.setMessage(resultCodeEnum.getMessage());
        return r;
    }

    /**
     * 功能描述：设置success状态
     *
     * @param success 值为true或false
     * @return R 返回给前端的数据
     */
    public R success(Boolean success) {
        this.setSuccess(success);
        return this;
    }

    /**
     * 功能描述：设置message内容
     *
     * @param message 消息内容
     * @return R 返回给前端的数据
     */
    public R message(String message) {
        this.setMessage(message);
        return this;
    }

    /**
     * 功能描述：设置code内容
     *
     * @param code 返回码
     * @return R 返回给前端的数据
     */
    public R code(Integer code) {
        this.setCode(code);
        return this;
    }

    /**
     * 功能描述：单值设置数据
     *
     * @param key   键
     * @param value 值
     * @return R 返回给前端的数据
     */
    public R data(String key, Object value) {
        this.data.put(key, value);
        return this;
    }

    /**
     * 功能描述：多值设置数据
     *
     * @param map 集合
     * @return R 返回给前端的数据
     */
    public R data(Map<String, Object> map) {
        this.setData(map);
        return this;
    }
}
